package com.rul.mars.user.service.impl;

import cn.hutool.core.bean.BeanUtil;
import cn.hutool.core.collection.CollUtil;
import com.github.pagehelper.PageHelper;
import com.rul.mars.common.constant.RoleStatusConstant;
import com.rul.mars.mbg.dao.RoleResourceRelationDao;
import com.rul.mars.mbg.domain.RoleParam;
import com.rul.mars.mbg.mapper.RoleMapper;
import com.rul.mars.mbg.mapper.RoleResourceRelationMapper;
import com.rul.mars.mbg.model.*;
import com.rul.mars.user.service.ResourceService;
import com.rul.mars.user.service.RoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;

import java.util.List;

/**
 * 角色服务实现
 *
 * @author RuL
 */
@Service
public class RoleServiceImpl implements RoleService {

    @Autowired
    private RoleMapper roleMapper;
    @Autowired
    private RoleResourceRelationMapper roleResourceRelationMapper;
    @Autowired
    private RoleResourceRelationDao roleResourceRelationDao;

    @Autowired
    private ResourceService resourceService;

    /**
     * 创建新角色
     *
     * @param param 角色信息
     * @return 插入行数
     */
    @Override
    public int create(RoleParam param) {
        Role role = BeanUtil.copyProperties(param, Role.class);
        //设置角色用户数量为0
        role.setUserCount(0L);
        //角色启用状态默认启用
        role.setStatus(RoleStatusConstant.ENABLE);
        return roleMapper.insertSelective(role);
    }


    /**
     * 修改角色信息
     *
     * @param role 角色信息
     * @return 修改行数
     */
    @Override
    public int update(Role role) {
        return roleMapper.updateByPrimaryKeySelective(role);
    }

    /**
     * 批量删除角色
     *
     * @param ids 需删除的角色ID
     * @return 删除角色数量
     */
    @Override
    public int delete(List<Long> ids) {
        RoleExample example = new RoleExample();
        example.createCriteria()
                .andIdIn(ids);
        int count = roleMapper.deleteByExample(example);
        resourceService.initResourceRolesMap();
        return count;
    }

    /**
     * 查询所有角色信息
     *
     * @return 所有角色信息
     */
    @Override
    public List<Role> list() {
        return roleMapper.selectByExample(new RoleExample());
    }

    /**
     * 分页获取角色信息
     *
     * @param pageSize 页面大小
     * @param pageNum  页号
     * @return 当前页角色信息
     */
    @Override
    public List<Role> list(Integer pageSize, Integer pageNum) {
        PageHelper.startPage(pageNum, pageSize);
        return roleMapper.selectByExample(new RoleExample());
    }

    /**
     * 根据角色ID获取所有资源信息
     *
     * @param roleId 角色ID
     * @return 该角色的所有资源信息
     */
    @Override
    public List<Resource> listResource(Long roleId) {
        return roleResourceRelationDao.getResourceListByRoleId(roleId);
    }

    /**
     * 为角色分配资源
     *
     * @param roleId      角色Id
     * @param resourceIds 资源Id列表
     * @return 分配数量
     */
    @Override
    public int allocResource(Long roleId, List<Long> resourceIds) {
        RoleResourceRelationExample example = new RoleResourceRelationExample();
        example.createCriteria()
                .andRoleIdEqualTo(roleId);
        //删除原有关系
        roleResourceRelationMapper.deleteByExample(example);
        for (Long resourceId : resourceIds) {
            RoleResourceRelation roleResourceRelation = new RoleResourceRelation();
            roleResourceRelation.setRoleId(roleId);
            roleResourceRelation.setResourceId(resourceId);

            roleResourceRelationMapper.insertSelective(roleResourceRelation);
        }
        resourceService.initResourceRolesMap();
        return resourceIds.size();
    }

    /**
     * 通过角色名查询角色信息
     *
     * @param name 角色名
     * @return 角色信息
     */
    @Override
    public Role getRoleByName(String name) {
        RoleExample example = new RoleExample();
        example.createCriteria()
                .andNameEqualTo(name);

        List<Role> roles = roleMapper.selectByExample(example);
        if (CollUtil.isNotEmpty(roles)) {
            return roles.get(0);
        }
        return null;
    }
}
